/**
 * @file lv_port_fs_templ.c
 *
 */

#ifndef _FS_DEV_H_
#define _FS_DEV_H_

/*Copy this file as "lv_port_fs.c" and set this value to "1" to enable content*/
#if 1

/*********************
 *      INCLUDES
 *********************/
#include "main.h"
/*********************
 *      DEFINES
 *********************/

/* Create a type to store the required data about your file.*/
typedef File file_t;

/*Similarly to `file_t` create a type for directory reading too */
typedef File rddir_t;

#define SPIFFS_FILE(file_p) ((file_t *)file_p)
#define SPIFFS_DIR(dir_p) ((rddir_t *)dir_p)

/**********************
 *      TYPEDEFS
 **********************/

/**********************
 *  STATIC PROTOTYPES
 **********************/
static void fs_init(void);

static void *fs_open(lv_fs_drv_t *drv, const char *path, lv_fs_mode_t mode);
static lv_fs_res_t fs_close(lv_fs_drv_t *drv, void *file_p);
static lv_fs_res_t fs_read(lv_fs_drv_t *drv, void *file_p, void *buf,
                           uint32_t btr, uint32_t *br);
static lv_fs_res_t fs_write(lv_fs_drv_t *drv, void *file_p, const void *buf,
                            uint32_t btw, uint32_t *bw);
static lv_fs_res_t fs_seek(lv_fs_drv_t *drv, void *file_p, uint32_t pos,
                           lv_fs_whence_t whence);
static lv_fs_res_t fs_size(lv_fs_drv_t *drv, void *file_p, uint32_t *size_p);
static lv_fs_res_t fs_tell(lv_fs_drv_t *drv, void *file_p, uint32_t *pos_p);

static void *fs_dir_open(lv_fs_drv_t *drv, const char *path);
static lv_fs_res_t fs_dir_read(lv_fs_drv_t *drv, void *rddir_p, char *fn);
static lv_fs_res_t fs_dir_close(lv_fs_drv_t *drv, void *rddir_p);

/**********************
 *  STATIC VARIABLES
 **********************/

/**********************
 * GLOBAL PROTOTYPES
 **********************/

/**********************
 *      MACROS
 **********************/

/**********************
 *   GLOBAL FUNCTIONS
 **********************/

void lv_port_fs_init(void) {
  /*----------------------------------------------------
   * Initialize your storage device and File System
   * -------------------------------------------------*/
  fs_init();

  /*---------------------------------------------------
   * Register the file system interface in LVGL
   *--------------------------------------------------*/

  /*Add a simple drive to open images*/
  static lv_fs_drv_t fs_drv;
  lv_fs_drv_init(&fs_drv);

  /*Set up fields...*/
  fs_drv.letter = 'F';
  fs_drv.open_cb = fs_open;
  fs_drv.close_cb = fs_close;
  fs_drv.read_cb = fs_read;
  fs_drv.write_cb = fs_write;
  fs_drv.seek_cb = fs_seek;
  fs_drv.tell_cb = fs_tell;

  fs_drv.dir_close_cb = fs_dir_close;
  fs_drv.dir_open_cb = fs_dir_open;
  fs_drv.dir_read_cb = fs_dir_read;

  lv_fs_drv_register(&fs_drv);
}

void listDir(fs::FS &fs, const char *dirname, uint8_t levels) {
  Serial.printf("Listing directory: %s\r\n", dirname);

  File root = fs.open(dirname);
  if (!root) {
    Serial.println("- failed to open directory");
    return;
  }
  if (!root.isDirectory()) {
    Serial.println(" - not a directory");
    return;
  }

  File file = root.openNextFile();
  while (file) {
    if (file.isDirectory()) {
      Serial.print("  DIR : ");
      Serial.println(file.name());
      if (levels) {
        // listDir(fs, file.path(), levels - 1);
      }
    } else {
      Serial.print("  FILE: ");
      Serial.print(file.name());
      Serial.print("\tSIZE: ");
      Serial.println(file.size());
    }
    file = root.openNextFile();
  }
}

/**********************
 *   STATIC FUNCTIONS
 **********************/

/*Initialize your Storage device and File system.*/
static void fs_init(void) {
  /*E.g. for FatFS initialize the SD card and FatFS itself*/
  if (!SPIFFS.begin()) {
    Serial.println("SPIFFS Mount Failed");
    return;
  } else {
    Serial.println("SPIFFS Init Sucess!");
  }

  listDir(SPIFFS, "/", 0);
  Serial.println("Test complete");
  /*You code here*/
}

/**
 * Open a file
 * @param drv       pointer to a driver where this function belongs
 * @param path      path to the file beginning with the driver letter (e.g.
 * S:/folder/file.txt)
 * @param mode      read: FS_MODE_RD, write: FS_MODE_WR, both: FS_MODE_RD |
 * FS_MODE_WR
 * @return          a file descriptor or NULL on error
 */
static void *fs_open(lv_fs_drv_t *drv, const char *path, lv_fs_mode_t mode) {
  String file_mode = "";

  if (mode == LV_FS_MODE_WR) {
    /*Open a file for write*/
    file_mode = FILE_WRITE;
  } else if (mode == LV_FS_MODE_RD) {
    /*Open a file for read*/
    file_mode = FILE_READ;
  } else if (mode == (LV_FS_MODE_WR | LV_FS_MODE_RD)) {
    /*Open a file for read and write*/
    file_mode = FILE_APPEND;
  }

  static file_t file = SPIFFS.open(path, file_mode.c_str());

  if (!file || file.isDirectory()) {
    LV_LOG_WARN("- failed to open file for reading");
    return NULL;
  }

  return &file;
}

/**
 * Close an opened file
 * @param drv       pointer to a driver where this function belongs
 * @param file_p    pointer to a file_t variable. (opened with lv_ufs_open)
 * @return          LV_FS_RES_OK: no error or  any error from @lv_fs_res_t enum
 */
static lv_fs_res_t fs_close(lv_fs_drv_t *drv, void *file_p) {
  lv_fs_res_t res = LV_FS_RES_OK;

  SPIFFS_FILE(file_p)->close();

  return res;
}

/**
 * Read data from an opened file
 * @param drv       pointer to a driver where this function belongs
 * @param file_p    pointer to a file_t variable.
 * @param buf       pointer to a memory block where to store the read data
 * @param btr       number of Bytes To Read
 * @param br        the real number of read bytes (Byte Read)
 * @return          LV_FS_RES_OK: no error or  any error from @lv_fs_res_t enum
 */
static lv_fs_res_t fs_read(lv_fs_drv_t *drv, void *file_p, void *buf,
                           uint32_t btr, uint32_t *br) {
  LV_LOG_WARN("read C:/liuzewen.txt ：%s \n", SPIFFS_FILE(file_p)->name());

  int ret = SPIFFS_FILE(file_p)->read((uint8_t *)buf, (size_t)btr);

  if (ret < 0) {
    return LV_FS_RES_FS_ERR;
  }

  *br = ret;

  return LV_FS_RES_OK;
}

/**
 * Write into a file
 * @param drv       pointer to a driver where this function belongs
 * @param file_p    pointer to a file_t variable
 * @param buf       pointer to a buffer with the bytes to write
 * @param btr       Bytes To Write
 * @param br        the number of real written bytes (Bytes Written). NULL if
 * unused.
 * @return          LV_FS_RES_OK: no error or  any error from @lv_fs_res_t enum
 */
static lv_fs_res_t fs_write(lv_fs_drv_t *drv, void *file_p, const void *buf,
                            uint32_t btw, uint32_t *bw) {
  /*Add your code here*/
  int ret = SPIFFS_FILE(file_p)->write((const uint8_t *)buf, (size_t)btw);

  if (ret < 0) {
    return LV_FS_RES_FS_ERR;
  }

  *bw = ret;

  return LV_FS_RES_OK;
}

/**
 * Set the read write pointer. Also expand the file size if necessary.
 * @param drv       pointer to a driver where this function belongs
 * @param file_p    pointer to a file_t variable. (opened with lv_ufs_open )
 * @param pos       the new position of read write pointer
 * @param whence    tells from where to interpret the `pos`. See @lv_fs_whence_t
 * @return          LV_FS_RES_OK: no error or  any error from @lv_fs_res_t enum
 */
static lv_fs_res_t fs_seek(lv_fs_drv_t *drv, void *file_p, uint32_t pos,
                           lv_fs_whence_t whence) {
  if (whence == LV_FS_SEEK_SET) {
    SPIFFS_FILE(file_p)->seek(pos, SeekSet);
  } else if (whence == LV_FS_SEEK_CUR) {
    SPIFFS_FILE(file_p)->seek(pos, SeekCur);
  } else if (whence == LV_FS_SEEK_END) {
    SPIFFS_FILE(file_p)->seek(pos, SeekEnd);
  } else {
    return LV_FS_RES_UNKNOWN;
  }

  return LV_FS_RES_OK;
}

static lv_fs_res_t fs_size(lv_fs_drv_t *drv, void *file_p, uint32_t *size_p) {
  return LV_FS_RES_OK;
}

/**
 * Give the position of the read write pointer
 * @param drv       pointer to a driver where this function belongs
 * @param file_p    pointer to a file_t variable.
 * @param pos_p     pointer to to store the result
 * @return          LV_FS_RES_OK: no error or  any error from @lv_fs_res_t enum
 */
static lv_fs_res_t fs_tell(lv_fs_drv_t *drv, void *file_p, uint32_t *pos_p) {
  *pos_p = SPIFFS_FILE(file_p)->position();
  return LV_FS_RES_OK;
}

/**
 * Initialize a 'lv_fs_dir_t' variable for directory reading
 * @param drv       pointer to a driver where this function belongs
 * @param path      path to a directory
 * @return          pointer to the directory read descriptor or NULL on error
 */
static void *fs_dir_open(lv_fs_drv_t *drv, const char *path) {
  rddir_t dir = SPIFFS.open(path);

  if (!dir) {
    Serial.println("- failed to open directory");
    return NULL;
  }
  if (!dir.isDirectory()) {
    Serial.println(" - not a directory");
    return NULL;
  }

  return &dir;
}

/**
 * Read the next filename form a directory.
 * The name of the directories will begin with '/'
 * @param drv       pointer to a driver where this function belongs
 * @param rddir_p   pointer to an initialized 'lv_fs_dir_t' variable
 * @param fn        pointer to a buffer to store the filename
 * @return          LV_FS_RES_OK: no error or  any error from @lv_fs_res_t enum
 */
static lv_fs_res_t fs_dir_read(lv_fs_drv_t *drv, void *rddir_p, char *fn) {
  String name;

  file_t file = SPIFFS_DIR(rddir_p)->openNextFile();

  do {
    if (SPIFFS_DIR(rddir_p)->openNextFile()) {
      name = file.name();

      if (file.isDirectory()) {
        fn[0] = '/';
        strcpy(&fn[1], name.c_str());
      } else {
        strcpy(fn, name.c_str());
      }

      file.close();
    } else {
      fn[0] = '\0';
    }
  } while (strcmp(fn, "/.") == 0 || strcmp(fn, "/..") == 0);

  return LV_FS_RES_OK;
}

/**
 * Close the directory reading
 * @param drv       pointer to a driver where this function belongs
 * @param rddir_p   pointer to an initialized 'lv_fs_dir_t' variable
 * @return          LV_FS_RES_OK: no error or  any error from @lv_fs_res_t enum
 */
static lv_fs_res_t fs_dir_close(lv_fs_drv_t *drv, void *rddir_p) {
  SPIFFS_DIR(rddir_p)->close();
  return LV_FS_RES_OK;
}

#else /*Enable this file at the top*/

/*This dummy typedef exists purely to silence -Wpedantic.*/
typedef int keep_pedantic_happy;
#endif

#endif
